# 导入所需库
from langchain_core.prompts import PromptTemplate
from langchain_deepseek import ChatDeepSeek
import os
from dotenv import load_dotenv, find_dotenv

from langchain.agents import create_openai_tools_agent, AgentExecutor, tool

# 1、使用.env 存储api_key
load_dotenv()

# 2、实例化ChatDeepSeek
chat = ChatDeepSeek(
    model="deepseek-chat",
    temperature=0.1, # 可选参数，控制生成结果的随机性
)

query = "今天我作了一个梦，梦里到处四面楚歌，我腹背受敌。最后我不得不冲出重围。"

prompt = PromptTemplate.from_template("根据内容提取1个关键词，只返回关键词，内容为:{topic}")
prompt_value = prompt.invoke({"topic": query})

keyword = chat.invoke(prompt_value)
print(keyword)

agent = create_openai_tools_agent(
    llm=chat,
    prompt="你好"
)

